import logging
from datetime import datetime

def setup_logger_with_timestamp(log_dir="logs", log_level=logging.INFO):
    """设置并返回一个日志器，日志文件名包含时间戳。

    Args:
        log_dir (str): 日志文件的目录，默认为'logs'。
        log_level (int): 日志级别，默认为logging.INFO。

    Returns:
        logging.Logger: 配置好的日志器实例。
    """
    # 获取当前时间并格式化为字符串，用于日志文件名
    now = datetime.now()
    timestamp = now.strftime("%Y-%m-%d_%H-%M-%S")
    log_file = f"{log_dir}/log_{timestamp}.log"

    # 确保日志目录存在
    if not os.path.exists(log_dir):
        os.makedirs(log_dir)

    # 创建一个logger
    logger = logging.getLogger(__name__)
    logger.setLevel(log_level)

    # 创建一个handler，用于写入日志文件
    file_handler = logging.FileHandler(log_file)
    file_handler.setLevel(log_level)

    # 定义handler的输出格式
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)

    # 给logger添加handler
    logger.addHandler(file_handler)

    return logger

def setup_logger(log_file, log_level=logging.INFO):

    """设置并返回一个日志器，用于向指定文件输出日志。
    Args:
        log_file (str): 日志文件的路径。

        log_level (int): 日志级别，默认为logging.INFO。
    Returns:
        logging.Logger: 配置好的日志器实例。
    """
    # 创建一个logger
    logger = logging.getLogger(__name__)
    logger.setLevel(log_level)

    # 创建一个handler，用于写入日志文件
    file_handler = logging.FileHandler(log_file)
    file_handler.setLevel(log_level)
    # 定义handler的输出格式
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)
    # 给logger添加handler
    logger.addHandler(file_handler)
    return logger

# 使用示例
def main():
    logger = setup_logger_with_timestamp()
    
    # 记录不同级别的日志
    logger.debug("This is a debug message.")
    logger.info("This is an info message.")
    logger.warning("This is a warning message.")
    logger.error("This is an error message.")
    logger.critical("This is a critical message.")

if __name__ == "__main__":
    import os
    main()